package stack;

import java.util.Stack;

/**
 * 150  逆波兰表达式求值
 * 给你一个字符串数组 tokens ，表示一个根据逆波兰表示法 表示的算术表达式。
 * 请你计算该表达式。返回一个表示表达式值的整数。
 *
 * 注意：
 * 有效的算符为 '+'、'-'、'*' 和 '/' 。
 * 每个操作数（运算对象）都可以是一个整数或者另一个表达式。
 * 两个整数之间的除法总是 向零截断 。
 * 表达式中不含除零运算。
 * 输入是一个根据逆波兰表示法表示的算术表达式。
 * 答案及所有中间计算结果可以用 32 位 整数表示。
 *
 * 逆波兰表达式：
 *
 * 逆波兰表达式是一种后缀表达式，所谓后缀就是指算符写在后面。
 *
 * 平常使用的算式则是一种中缀表达式，如 ( 1 + 2 ) * ( 3 + 4 ) 。
 * 该算式的逆波兰表达式写法为 ( ( 1 2 + ) ( 3 4 + ) * ) 。
 * 逆波兰表达式主要有以下两个优点：
 *
 * 去掉括号后表达式无歧义，上式即便写成 1 2 + 3 4 + * 也可以依据次序计算出正确结果。
 * 适合用栈操作运算：遇到数字则入栈；遇到算符则取出栈顶两个数字进行计算，并将结果压入栈中
 *
 */


// 遇到数字则入栈；遇到算符则取出栈顶两个数字进行计算，并将结果压入栈中
public class EvaluateReversePolishNonation_150 {
    public static int evalRPN(String[] tokens) {
        Stack<Integer> stack = new Stack<>();
        // token 的长度可能为1
        if (tokens.length == 1) {
            return Integer.parseInt(tokens[0]);
        }
        int res = 0;
        for (String token : tokens) {
            if ("+".equals(token)) {
                int n1 = stack.pop();
                int n2 = stack.pop();
                res = n2 + n1;
                stack.add(res);
            } else if ("-".equals(token)) {
                int n1 = stack.pop();
                int n2 = stack.pop();
                res = n2 - n1;
                stack.add(res);
            } else if ("*".equals(token)) {
                int n1 = stack.pop();
                int n2 = stack.pop();
                res = n2 * n1;
                stack.add(res);
            } else if ("/".equals(token)) {
                int n1 = stack.pop();
                int n2 = stack.pop();
                res = n2 / n1;
                stack.add(res);
            } else {
                stack.add(Integer.parseInt(token));
            }
            System.out.println("stack: " + stack);
        }
        return res;
    }

    public static int evalRPN2(String[] tokens) {
        Stack<Integer> stack = new Stack<>();
        int res = 0;
        for (String token : tokens) {
            if ("+".equals(token)) {
                int n1 = stack.pop();
                int n2 = stack.pop();
                res = n2 + n1;
                stack.add(res);
            } else if ("-".equals(token)) {
                int n1 = stack.pop();
                int n2 = stack.pop();
                res = n2 - n1;
                stack.add(res);
            } else if ("*".equals(token)) {
                int n1 = stack.pop();
                int n2 = stack.pop();
                res = n2 * n1;
                stack.add(res);
            } else if ("/".equals(token)) {
                int n1 = stack.pop();
                int n2 = stack.pop();
                res = n2 / n1;
                stack.add(res);
            } else {
                stack.add(Integer.parseInt(token));
            }
            System.out.println("stack: " + stack);
        }

        // 当tokens长度为1时，也可以直接pop，反正最后res也进入了stack，也可以pop出去
        return stack.pop();
    }

    public static void main(String[] args) {
        String[] tokens = new String[]{"4", "13", "5", "/", "+"};
        System.out.println(evalRPN(tokens));
    }

}
